from modules.geometry import rotation_matrix_to_axis_angle
from scipy.spatial.transform import Rotation as R
from utils import path_util, get_input
import joblib
import json
import numpy as np
import os
import metric
import torch


def eval_vibe():
    for idx in [41]:
        metric.output_metric(get_input.get_vibe_poses(idx),
                             get_input.get_gt_poses(idx))


def eval_hmr():
    for idx in [7, 24, 29, 41]:
        metric.output_metric(get_input.get_hmr_poses(idx),
                             get_input.get_gt_poses(idx))


def eval(name):
    for idx in [7, 24, 29, 41]:
        pred_poses = get_input.get_pred_poses(name, idx)
        gt_poses = get_input.get_gt_poses(idx)
        pred_poses = pred_poses[:len(gt_poses)]
        metric.output_metric(pred_poses, gt_poses)


if __name__ == '__main__':
    name = input()
    eval(name)
